import os


def file_size(dir):
    '''
    统计文件大小
    :return:
    '''
    sum = 0

    # 判断目录是否为文件,如果是直接读取大小
    if os.path.isfile(dir):
        sum += os.path.getsize(dir)
    # 如果不是文件，先判断是否为目录
    elif os.path.isdir(dir):
        dirlist = os.listdir()  # 返回目录下所有文件和目录名
        for f in dirlist:
            file = os.path.join(dirlist, f)  # 拼接路径
            if os.path.isfile(file):
                sum += os.path.getsize(file)
            if os.path.isdir(file):
                sum += file_size(file)  # 递归调用

    return sum


print(file_size('Test.py'))
